home *** CD-ROM | disk | FTP | other *** search
/ Mac Easy 2010 May / Mac Life Ubuntu.iso / casper / filesystem.squashfs / usr / lib / perl / 5.10.0 / IO / Socket / INET.pm next >
Encoding:
Perl POD Document  |  2009-06-26  |  7.3 KB  |  312 lines
  1. # IO::Socket::INET.pm
  2. #
  3. # Copyright (c) 1997-8 Graham Barr <gbarr@pobox.com>. All rights reserved.
  4. # This program is free software; you can redistribute it and/or
  5. # modify it under the same terms as Perl itself.
  6.  
  7. package IO::Socket::INET;
  8.  
  9. use strict;
  10. our(@ISA, $VERSION);
  11. use IO::Socket;
  12. use Socket;
  13. use Carp;
  14. use Exporter;
  15. use Errno;
  16.  
  17. @ISA = qw(IO::Socket);
  18. $VERSION = "1.31";
  19.  
  20. my $EINVAL = exists(&Errno::EINVAL) ? Errno::EINVAL() : 1;
  21.  
  22. IO::Socket::INET->register_domain( AF_INET );
  23.  
  24. my %socket_type = ( tcp  => SOCK_STREAM,
  25.             udp  => SOCK_DGRAM,
  26.             icmp => SOCK_RAW
  27.           );
  28. my %proto_number;
  29. $proto_number{tcp}  = Socket::IPPROTO_TCP()  if defined &Socket::IPPROTO_TCP;
  30. $proto_number{udp}  = Socket::IPPROTO_UDP()  if defined &Socket::IPPROTO_UDP;
  31. $proto_number{icmp} = Socket::IPPROTO_ICMP() if defined &Socket::IPPROTO_ICMP;
  32. my %proto_name = reverse %proto_number;
  33.  
  34. sub new {
  35.     my $class = shift;
  36.     unshift(@_, "PeerAddr") if @_ == 1;
  37.     return $class->SUPER::new(@_);
  38. }
  39.  
  40. sub _cache_proto {
  41.     my @proto = @_;
  42.     for (map lc($_), $proto[0], split(' ', $proto[1])) {
  43.     $proto_number{$_} = $proto[2];
  44.     }
  45.     $proto_name{$proto[2]} = $proto[0];
  46. }
  47.  
  48. sub _get_proto_number {
  49.     my $name = lc(shift);
  50.     return undef unless defined $name;
  51.     return $proto_number{$name} if exists $proto_number{$name};
  52.  
  53.     my @proto = getprotobyname($name);
  54.     return undef unless @proto;
  55.     _cache_proto(@proto);
  56.  
  57.     return $proto[2];
  58. }
  59.  
  60. sub _get_proto_name {
  61.     my $num = shift;
  62.     return undef unless defined $num;
  63.     return $proto_name{$num} if exists $proto_name{$num};
  64.  
  65.     my @proto = getprotobynumber($num);
  66.     return undef unless @proto;
  67.     _cache_proto(@proto);
  68.  
  69.     return $proto[0];
  70. }
  71.  
  72. sub _sock_info {
  73.   my($addr,$port,$proto) = @_;
  74.   my $origport = $port;
  75.   my @serv = ();
  76.  
  77.   $port = $1
  78.     if(defined $addr && $addr =~ s,:([\w\(\)/]+)$,,);
  79.  
  80.   if(defined $proto  && $proto =~ /\D/) {
  81.     my $num = _get_proto_number($proto);
  82.     unless (defined $num) {
  83.       $@ = "Bad protocol '$proto'";
  84.       return;
  85.     }
  86.     $proto = $num;
  87.   }
  88.  
  89.   if(defined $port) {
  90.     my $defport = ($port =~ s,\((\d+)\)$,,) ? $1 : undef;
  91.     my $pnum = ($port =~ m,^(\d+)$,)[0];
  92.  
  93.     @serv = getservbyname($port, _get_proto_name($proto) || "")
  94.     if ($port =~ m,\D,);
  95.  
  96.     $port = $serv[2] || $defport || $pnum;
  97.     unless (defined $port) {
  98.     $@ = "Bad service '$origport'";
  99.     return;
  100.     }
  101.  
  102.     $proto = _get_proto_number($serv[3]) if @serv && !$proto;
  103.   }
  104.  
  105.  return ($addr || undef,
  106.      $port || undef,
  107.      $proto || undef
  108.     );
  109. }
  110.  
  111. sub _error {
  112.     my $sock = shift;
  113.     my $err = shift;
  114.     {
  115.       local($!);
  116.       my $title = ref($sock).": ";
  117.       $@ = join("", $_[0] =~ /^$title/ ? "" : $title, @_);
  118.       $sock->close()
  119.     if(defined fileno($sock));
  120.     }
  121.     $! = $err;
  122.     return undef;
  123. }
  124.  
  125. sub _get_addr {
  126.     my($sock,$addr_str, $multi) = @_;
  127.     my @addr;
  128.     if ($multi && $addr_str !~ /^\d+(?:\.\d+){3}$/) {
  129.     (undef, undef, undef, undef, @addr) = gethostbyname($addr_str);
  130.     } else {
  131.     my $h = inet_aton($addr_str);
  132.     push(@addr, $h) if defined $h;
  133.     }
  134.     @addr;
  135. }
  136.  
  137. sub configure {
  138.     my($sock,$arg) = @_;
  139.     my($lport,$rport,$laddr,$raddr,$proto,$type);
  140.  
  141.     $arg->{LocalAddr} = $arg->{LocalHost}
  142.     if exists $arg->{LocalHost} && !exists $arg->{LocalAddr};
  143.  
  144.     ($laddr,$lport,$proto) = _sock_info($arg->{LocalAddr},
  145.                     $arg->{LocalPort},
  146.                     $arg->{Proto})
  147.             or return _error($sock, $!, $@);
  148.  
  149.     $laddr = defined $laddr ? inet_aton($laddr)
  150.                 : INADDR_ANY;
  151.  
  152.     return _error($sock, $EINVAL, "Bad hostname '",$arg->{LocalAddr},"'")
  153.     unless(defined $laddr);
  154.  
  155.     $arg->{PeerAddr} = $arg->{PeerHost}
  156.     if exists $arg->{PeerHost} && !exists $arg->{PeerAddr};
  157.  
  158.     unless(exists $arg->{Listen}) {
  159.     ($raddr,$rport,$proto) = _sock_info($arg->{PeerAddr},
  160.                         $arg->{PeerPort},
  161.                         $proto)
  162.             or return _error($sock, $!, $@);
  163.     }
  164.  
  165.     $proto ||= _get_proto_number('tcp');
  166.  
  167.     $type = $arg->{Type} || $socket_type{lc _get_proto_name($proto)};
  168.  
  169.     my @raddr = ();
  170.  
  171.     if(defined $raddr) {
  172.     @raddr = $sock->_get_addr($raddr, $arg->{MultiHomed});
  173.     return _error($sock, $EINVAL, "Bad hostname '",$arg->{PeerAddr},"'")
  174.         unless @raddr;
  175.     }
  176.  
  177.     while(1) {
  178.  
  179.     $sock->socket(AF_INET, $type, $proto) or
  180.         return _error($sock, $!, "$!");
  181.  
  182.         if (defined $arg->{Blocking}) {
  183.         defined $sock->blocking($arg->{Blocking})
  184.         or return _error($sock, $!, "$!");
  185.     }
  186.  
  187.     if ($arg->{Reuse} || $arg->{ReuseAddr}) {
  188.         $sock->sockopt(SO_REUSEADDR,1) or
  189.             return _error($sock, $!, "$!");
  190.     }
  191.  
  192.     if ($arg->{ReusePort}) {
  193.         $sock->sockopt(SO_REUSEPORT,1) or
  194.             return _error($sock, $!, "$!");
  195.     }
  196.  
  197.     if ($arg->{Broadcast}) {
  198.         $sock->sockopt(SO_BROADCAST,1) or
  199.             return _error($sock, $!, "$!");
  200.     }
  201.  
  202.     if($lport || ($laddr ne INADDR_ANY) || exists $arg->{Listen}) {
  203.         $sock->bind($lport || 0, $laddr) or
  204.             return _error($sock, $!, "$!");
  205.     }
  206.  
  207.     if(exists $arg->{Listen}) {
  208.         $sock->listen($arg->{Listen} || 5) or
  209.         return _error($sock, $!, "$!");
  210.         last;
  211.     }
  212.  
  213.      # don't try to connect unless we're given a PeerAddr
  214.      last unless exists($arg->{PeerAddr});
  215.  
  216.         $raddr = shift @raddr;
  217.  
  218.     return _error($sock, $EINVAL, 'Cannot determine remote port')
  219.         unless($rport || $type == SOCK_DGRAM || $type == SOCK_RAW);
  220.  
  221.     last
  222.         unless($type == SOCK_STREAM || defined $raddr);
  223.  
  224.     return _error($sock, $EINVAL, "Bad hostname '",$arg->{PeerAddr},"'")
  225.         unless defined $raddr;
  226.  
  227. #        my $timeout = ${*$sock}{'io_socket_timeout'};
  228. #        my $before = time() if $timeout;
  229.  
  230.     undef $@;
  231.         if ($sock->connect(pack_sockaddr_in($rport, $raddr))) {
  232. #            ${*$sock}{'io_socket_timeout'} = $timeout;
  233.             return $sock;
  234.         }
  235.  
  236.     return _error($sock, $!, $@ || "Timeout")
  237.         unless @raddr;
  238.  
  239. #    if ($timeout) {
  240. #        my $new_timeout = $timeout - (time() - $before);
  241. #        return _error($sock,
  242. #                         (exists(&Errno::ETIMEDOUT) ? Errno::ETIMEDOUT() : $EINVAL),
  243. #                         "Timeout") if $new_timeout <= 0;
  244. #        ${*$sock}{'io_socket_timeout'} = $new_timeout;
  245. #        }
  246.  
  247.     }
  248.  
  249.     $sock;
  250. }
  251.  
  252. sub connect {
  253.     @_ == 2 || @_ == 3 or
  254.        croak 'usage: $sock->connect(NAME) or $sock->connect(PORT, ADDR)';
  255.     my $sock = shift;
  256.     return $sock->SUPER::connect(@_ == 1 ? shift : pack_sockaddr_in(@_));
  257. }
  258.  
  259. sub bind {
  260.     @_ == 2 || @_ == 3 or
  261.        croak 'usage: $sock->bind(NAME) or $sock->bind(PORT, ADDR)';
  262.     my $sock = shift;
  263.     return $sock->SUPER::bind(@_ == 1 ? shift : pack_sockaddr_in(@_))
  264. }
  265.  
  266. sub sockaddr {
  267.     @_ == 1 or croak 'usage: $sock->sockaddr()';
  268.     my($sock) = @_;
  269.     my $name = $sock->sockname;
  270.     $name ? (sockaddr_in($name))[1] : undef;
  271. }
  272.  
  273. sub sockport {
  274.     @_ == 1 or croak 'usage: $sock->sockport()';
  275.     my($sock) = @_;
  276.     my $name = $sock->sockname;
  277.     $name ? (sockaddr_in($name))[0] : undef;
  278. }
  279.  
  280. sub sockhost {
  281.     @_ == 1 or croak 'usage: $sock->sockhost()';
  282.     my($sock) = @_;
  283.     my $addr = $sock->sockaddr;
  284.     $addr ? inet_ntoa($addr) : undef;
  285. }
  286.  
  287. sub peeraddr {
  288.     @_ == 1 or croak 'usage: $sock->peeraddr()';
  289.     my($sock) = @_;
  290.     my $name = $sock->peername;
  291.     $name ? (sockaddr_in($name))[1] : undef;
  292. }
  293.  
  294. sub peerport {
  295.     @_ == 1 or croak 'usage: $sock->peerport()';
  296.     my($sock) = @_;
  297.     my $name = $sock->peername;
  298.     $name ? (sockaddr_in($name))[0] : undef;
  299. }
  300.  
  301. sub peerhost {
  302.     @_ == 1 or croak 'usage: $sock->peerhost()';
  303.     my($sock) = @_;
  304.     my $addr = $sock->peeraddr;
  305.     $addr ? inet_ntoa($addr) : undef;
  306. }
  307.  
  308. 1;
  309.  
  310. __END__
  311.  
  312.